package com.allenz.forum.controller;

import com.allenz.forum.common.AppResult;
import com.allenz.forum.common.ResultCode;
import com.allenz.forum.config.AppConfig;
import com.allenz.forum.model.Message;
import com.allenz.forum.model.User;
import com.allenz.forum.services.IMessageService;
import com.allenz.forum.services.IUserService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import lombok.NonNull;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
import java.util.List;

@Slf4j
@Api(tags = "站内信接口")
@RestController
@RequestMapping("/message")
public class MessageController {

    @Resource
    private IUserService userService;
    @Resource
    private IMessageService messageService;

    @ApiOperation("发送站内信")
    @PostMapping("/send")
    public AppResult send(HttpServletRequest request ,
                          @ApiParam("接收者id") @RequestParam("receiveUserId") @NonNull Long receiveUserId ,
                          @ApiParam("站内信内容") @RequestParam("content") @NonNull String content){
        // 1. 当前用户的状态 , 如果是禁言 , 则不能发送站内信
        HttpSession session = request.getSession(false);
        User user = (User) session.getAttribute(AppConfig.USER_SESSION);
        if (user.getState() == 1){
            // 返回用户状态异常
            return AppResult.failed(ResultCode.FAILED_USER_BANNED);
        }
        // 2. 不能给自己发送站内信
        if (user.getId() == receiveUserId){
            // 返回错误描述
            return AppResult.failed("不能给自己发送站内信!");
        }
        // 3. 校验接收者是否存在
        User receiveUser = userService.selectById(receiveUserId);
        if (receiveUser == null || receiveUser.getDeleteState() == 1){
            // 返回接收者状态不正常
            return AppResult.failed("接收者状态异常!");
        }
        // 4. 封装对象
        Message message = new Message();
        message.setPostUserId(user.getId()); // 发送者id
        message.setReceiveUserId(receiveUserId); // 接收者id
        message.setContent(content); // 站内信内容
        // 5. 调用 Service
        messageService.create(message);
        // 6. 返回结果
        return AppResult.success("发送成功");
    }

    @ApiOperation("获取未读数")
    @GetMapping("/getUnreadCount")
    public AppResult<Integer> getUnreadCount(HttpServletRequest request){
        // 1. 获取当前登录的用户
        HttpSession session = request.getSession(false);
        User user = (User) session.getAttribute(AppConfig.USER_SESSION);
        // 2. 调用 Service
        Integer count = messageService.selectUnreadCount(user.getId());
        // 3. 返回结果
        return AppResult.success(count);
    }

    @ApiOperation("查询用户所有站内信")
    @GetMapping("/getAll")
    public AppResult<List<Message>> getAll(HttpServletRequest request){
        // 1. 获取当前登录用户
        HttpSession session = request.getSession(false);
        User user = (User) session.getAttribute(AppConfig.USER_SESSION);
        // 2. 调用 Service
        List<Message> messages = messageService.selectByReceiveUserId(user.getId());
        // 3. 返回结果
        return AppResult.success(messages);
    }

    @ApiOperation("更新为已读")
    @PostMapping("/markRead")
    public AppResult markRead(HttpServletRequest request ,
                              @ApiParam("站内信id") @RequestParam("id") @NonNull Long id){
        // 根据id查询站内信
        Message message = messageService.selectById(id);
        // 站内信是否存在
        if (message == null || message.getDeleteState() == 1){
            // 返回错误描述
            return AppResult.failed(ResultCode.FAILED_MESSAGE_NOT_EXISTS);
        }
        // 站内信是不是自己的
        HttpSession session = request.getSession(false);
        User user = (User) session.getAttribute(AppConfig.USER_SESSION);
        if (user.getId() != message.getReceiveUserId()){
            // 返回错误描述
            return AppResult.failed(ResultCode.FAILED_FORBIDDEN);
        }
        // 调用 Service , 把状态更新为已读
        messageService.updateStateById(id , (byte) 1);
        //返回结果
        return AppResult.success();
    }

    /**
     * 回复站内信
     * @param repliedId 要回复的站内信Id
     * @param content 站内信内容
     * @return
     */
    @ApiOperation("回复站内信")
    @PostMapping("/reply")
    public AppResult reply(HttpServletRequest request ,
                           @ApiParam("要回复的站内信Id") @RequestParam("repliedId") @NonNull Long repliedId ,
                           @ApiParam("站内信内容") @RequestParam("content") @NonNull String content){
        // 校验当前用户状态
        HttpSession session = request.getSession(false);
        User user = (User) session.getAttribute(AppConfig.USER_SESSION);
        if (user.getState() == 1){
            // 返回错误描述
            return AppResult.failed(ResultCode.FAILED_USER_BANNED);
        }

        // 校验要回复的站内信状态
        Message existsMessage = messageService.selectById(repliedId);
        if (existsMessage == null || existsMessage.getDeleteState() == 1){
            // 返回错误描述
            return AppResult.failed(ResultCode.FAILED_MESSAGE_NOT_EXISTS);
        }
        // 不能给自己的站内信回复
        if (user.getId() == existsMessage.getPostUserId()){
            // 返回错误描述
            return AppResult.failed("不能回复自己的站内信");
        }

        // 构造对象
        Message message = new Message();
        message.setPostUserId(user.getId());
        message.setReceiveUserId(existsMessage.getPostUserId());
        message.setContent(content);
        // 调用Service
        messageService.reply(repliedId , message);
        // 返回结果
        return AppResult.success();
    }
}
